// TortoiseGit - a Windows shell extension for easy version control

// Copyright (C) 2013 - TortoiseGit
// Copyright (C) 2006, 2008 - TortoiseSVN

// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software Foundation,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//
#include "stdafx.h"
#include "DropFiles.h"

CDropFiles::CDropFiles()
{
	m_pBuffer = NULL;
	m_nBufferSize = 0;
}

CDropFiles::~CDropFiles()
{
	delete [] m_pBuffer;
}

void CDropFiles::AddFile(const CString &sFile)
{
	m_arFiles.Add(sFile);
}

INT_PTR CDropFiles::GetCount() const
{
	return m_arFiles.GetCount();
}

bool CDropFiles::IsEmpty() const
{
	return m_arFiles.IsEmpty() == TRUE;
}

void CDropFiles::CreateBuffer()
{
	ASSERT(m_pBuffer == NULL);
	ASSERT(m_nBufferSize == 0);
	ASSERT(!m_arFiles.IsEmpty());

	int nLength = 0;

	for(int i=0;i<m_arFiles.GetSize();i++)
	{
		nLength += m_arFiles[i].GetLength();
		nLength += 1; // '\0' separator
	}

	m_nBufferSize = sizeof(DROPFILES) + (nLength+1)*sizeof(TCHAR);
	m_pBuffer = new char[m_nBufferSize];

	SecureZeroMemory(m_pBuffer, m_nBufferSize);

	DROPFILES* df = (DROPFILES*)m_pBuffer;
	df->pFiles = sizeof(DROPFILES);
	df->fWide = 1;

	TCHAR* pFilenames = (TCHAR*)(m_pBuffer + sizeof(DROPFILES));
	TCHAR* pCurrentFilename = pFilenames;

	for(int i=0;i<m_arFiles.GetSize();i++)
	{
		CString str = m_arFiles[i];
		wcscpy_s(pCurrentFilename,str.GetLength()+1,str.GetBuffer());
		pCurrentFilename += str.GetLength();
		*pCurrentFilename = '\0'; // separator between file names
		pCurrentFilename++;
	}
	*pCurrentFilename = '\0'; // terminate array
}

void* CDropFiles::GetBuffer() const
{
	return (void*)m_pBuffer;
}

int	CDropFiles::GetBufferSize() const
{
	return m_nBufferSize;
}

void CDropFiles::CreateStructure()
{
	CreateBuffer();

	COleDataSource dropData;
	HGLOBAL hMem = ::GlobalAlloc(GMEM_ZEROINIT|GMEM_MOVEABLE|GMEM_DDESHARE, GetBufferSize());
	memcpy( ::GlobalLock(hMem), GetBuffer(), GetBufferSize() );
	::GlobalUnlock(hMem);
	dropData.CacheGlobalData( CF_HDROP, hMem );
	dropData.DoDragDrop(DROPEFFECT_COPY|DROPEFFECT_MOVE|DROPEFFECT_LINK,NULL);
}
